package com.ruoyi.common.dto;

public class TrainingRequest {
    private String configPath;
    private String dataPath;
    private String outputPath;
    private String modelName;
    private String device; // GPU or CPU

    public String getConfigPath() { return configPath; }
    public void setConfigPath(String configPath) { this.configPath = configPath; }

    public String getDataPath() { return dataPath; }
    public void setDataPath(String dataPath) { this.dataPath = dataPath; }

    public String getOutputPath() { return outputPath; }
    public void setOutputPath(String outputPath) { this.outputPath = outputPath; }

    public String getModelName() { return modelName; }
    public void setModelName(String modelName) { this.modelName = modelName; }

    public String getDevice() { return device; }
    public void setDevice(String device) { this.device = device; }
}
